import numpy as np


a = np.arange(8).reshape(2,2,2)
b = np.arange(4).reshape(2,2)
print(a)
print(b)
